"""
编写人：caoyong，时间：2022-11-23
该模块需要安装  pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple pymysql
"""
import pymysql
from caoyongpy import dbsql


class MySql(dbsql.SqlData):
    def __init__(self, host, port, user, passwd, db, charset='utf8'):
        """
        初始化数据库连接
        :param host: mysql连接地址
        :param port: 端口
        :param user:  用户名
        :param passwd:  密码
        :param db:  数据库名
        :param charset:  字符集编码
        """
        self.db = pymysql.connect(host=host, port=port, user=user, passwd=passwd, db=db, charset=charset)  # 打开数据库连接
        dbsql.SqlData.__init__(self, self.db, {"'": "''", "\\": "\\\\"})

    def close(self):  # 关闭数据库
        self.db.close()

    def commit(self):  # 提交事务
        self.db.commit()

    def rollback(self):  # 回滚事务
        self.db.rollback()




